from selenium import webdriver
from selenium.webdriver.common.by import By


# 可以放在类里面  也可以放在类外面   放在类里面要创建一个对象之后才能调用这个方法
def get_msg():
    driver=UtilsDriver.get_driver()
    driver.switch_to.frame("layui-layer-iframe1")
    msg=driver.find_element(By.CSS_SELECTOR, "[class='conect-title'] span").text
    return msg


class UtilsDriver:
    _driver=None  #_driver是一个私有的属性  不被其他类调用

    # 定义一个获取浏览器驱动对象的方法
    @classmethod    # 表示这个方法是一个类级别的方法
    def get_driver(cls):
        if cls._driver is None: # 判断driver是否已创建，如果为None就需要去创建浏览器驱动对象，如果不为None，则直接返回driver
            cls._driver = webdriver.Chrome()
            cls._driver.maximize_window()
            cls._driver.implicitly_wait(10)
            cls._driver.get("http://localhost/")
        return cls._driver

    # 定义一个退出浏览器驱动对象的方法
    @classmethod
    def quit_driver(cls):
        if cls._driver is not None:
            cls._driver.quit()
            cls._driver=None

